import java.util.*;

/**
 * @author 03010570
 * @date 2020/07/13
 * describe:    LeetCode : 350、两个数组的交集2 https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/
 */
public class LeetCode_350 {

    public static void main(String[] args) {
        int[] nums1 = {1, 2, 2, 1};
        int[] nums2 = {2, 2};
        int[] intersection = intersect(nums1, nums2);
        for (int i : intersection) {
            System.out.print(i + " , ");
        }
        System.out.println();
        int[] nums3 = {4, 9, 5};
        int[] nums4 = {9, 4, 9, 8, 4};
        int[] intersection2 = intersect(nums3, nums4);
        for (int i : intersection2) {
            System.out.print(i + " , ");
        }
    }

    /**
     * 排序数组比较法：
     * 循环两个数组，从 左边依次开始比较。
     * 两个值相等，添加到返回的数组当中，两个数组的下标++
     * 哪一个数组的值较小，下标 ++；
     * 时间复杂度： O(M+N)
     * 空间复杂度； O(min(M,N))
     *
     * @param nums1
     * @param nums2
     * @return
     */
    public static int[] intersect(int[] nums1, int[] nums2) {
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int length1 = nums1.length;
        int length2 = nums2.length;
        int index = 0, index1 = 0, index2 = 0;
        int[] res = new int[Math.min(length1, length2)];
        while (index1 < length1 && index2 < length2) {
            if (nums1[index1] < nums2[index2]) {
                index1++;
            } else if (nums1[index1] > nums2[index2]) {
                index2++;
            } else {
                res[index++] = nums1[index1];
                index1++;
                index2++;
            }
        }
        return Arrays.copyOfRange(res, 0, index);

    }

    /**
     * 哈希表法 2 ：
     * 利用 HashMap 存储两个数组的值，在循环第二个数组时把子集添加到数组当中去。
     * 时间复杂度 ： O(M+N)  M 数组1的长度，N 数组2的长度
     * 空间复杂度 ： O(M)    数组1的长度
     *
     * @param nums1
     * @param nums2
     * @return
     */
    public static int[] intersect2(int[] nums1, int[] nums2) {
        if (nums1.length == 0 || nums2.length == 0) {
            return new int[0];
        }
        Map<Integer, Integer> map = new HashMap<>();
        for (int i : nums1) {
            map.put(i, map.getOrDefault(i, 0) + 1);
        }
        int index = 0;
        int[] res = new int[nums1.length];
        for (int i2 : nums2) {
            if (map.get(i2) != null && map.get(i2) != 0) {
                map.put(i2, map.get(i2) - 1);
                res[index++] = i2;
            }
        }
        return Arrays.copyOfRange(res, 0, index);

    }

    /**
     * 哈希表法1 ： 使用 HashMap 来存储两个数组的值，子集用 List 来存储，再转化成 数组 返回
     * 时间复杂度 ： O(M+N+K)    M 数组1的长度，N 数组2的长度，K 子集的长度
     * 空间复杂度 ： O(MAX(M,N)) 数组1的长度，但不知道是长的数组还是短的数组
     *
     * @param nums1
     * @param nums2
     * @return
     */

    public static int[] intersect1(int[] nums1, int[] nums2) {
        if (nums1.length == 0 || nums2.length == 0) {
            return new int[0];
        }
        Map<Integer, Integer> map = new HashMap<>();
        for (int i : nums1) {
            map.put(i, map.getOrDefault(i, 0) + 1);
        }
        List<Integer> list = new ArrayList<>();
        for (int i2 : nums2) {
            if (map.get(i2) != null && map.get(i2) != 0) {
                map.put(i2, map.get(i2) - 1);
                list.add(i2);
            }
        }
        int[] res = new int[list.size()];
        int index = 0;
        for (Integer i : list) {
            res[index++] = i;
        }
        return res;

    }

}
